A GPU card can be configured in one of two modes: vSGA (shared virtual graphics) and vGPU. The NVIDIA card should be configured with vGPU mode. This is specifically for use of the GPU in compute workloads, such as in machine learning or high performance computing applications. To enable vGPU mode on the ESXi host, use the command line to execute this command:
esxcli graphics host set --default-type SharedPassthru
Check the Host Graphics Settings